package com.itheima.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.util.Date;
import java.util.Map;

/**
 * JWT工具类，用于生成和校验JWT令牌
 */
public class JwtUtils {

    // 签名密钥，应妥善保管，建议从配置文件或环境变量获取
    private static final String SECRET_KEY = "itheima";

    // 令牌有效期（毫秒），默认1小时
    private static final long EXPIRATION_TIME = 1000 * 60 * 60;

    /**
     * 生成JWT令牌
     * @param claims 要包含在JWT中的用户信息
     * @return JWT字符串
     */
    public static String generateToken(Map<String, Object> claims) {
        return Jwts.builder()
                .setClaims(claims)
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
                .signWith(SignatureAlgorithm.HS256, SECRET_KEY)
                .compact();
    }

    /**
     * 校验JWT令牌并获取Claims
     * @param token JWT字符串
     * @return 解析后的Claims对象
     * @throws Exception 当令牌无效或过期时抛出异常
     */
    public static Claims validateToken(String token) throws Exception {
        return Jwts.parser()
                .setSigningKey(SECRET_KEY)
                .parseClaimsJws(token)
                .getBody();
    }

    /**
     * 从令牌中获取用户ID
     * @param token JWT字符串
     * @return 用户ID
     */
    public static Integer getUserId(String token) {
        try {
            Claims claims = validateToken(token);
            return claims.get("id", Integer.class);
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * 从令牌中获取用户名
     * @param token JWT字符串
     * @return 用户名
     */
    public static String getUsername(String token) {
        try {
            Claims claims = validateToken(token);
            return claims.get("username", String.class);
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * 检查令牌是否过期
     * @param token JWT字符串
     * @return true表示已过期，false表示未过期
     */
    public static boolean isTokenExpired(String token) {
        try {
            Claims claims = validateToken(token);
            return claims.getExpiration().before(new Date());
        } catch (Exception e) {
            return true;
        }
    }
}